import numpy as np


def print_np():
    # print('hello test')
    # 创建一个2x3的数组
    arr = np.array([[1, 2, 3], [4, 5, 6]])
    # 计算数组中所有元素的和
    print(np.sum(arr))  # 输出21
    # 计算数组中每行的和
    print(np.sum(arr, axis=1))  # 输出[5 7 9]
    # 计算数组中每列的和
    print(np.sum(arr, axis=0))  # 输出[10 15]


print_np()
